package com.iking.toolkit.ocr.service.impl;

import cn.hutool.http.HttpRequest;
import cn.hutool.http.HttpUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.esotericsoftware.minlog.Log;
import com.iking.toolkit.model.Msg;
import com.iking.toolkit.ocr.model.dto.ExtractUrlTextDTO;
import com.iking.toolkit.ocr.service.OcrService;
import com.iking.toolkit.redis.RedisObjectService;
import com.iking.toolkit.system.dao.DictionaryItemsMapper;
import com.iking.toolkit.system.model.model.DictionaryItems;
import com.iking.toolkit.util.ApiUrlUtil;
import com.iking.toolkit.util.FileFormatUtil;
import com.iking.toolkit.util.OCRUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.HashMap;

/**
 * 功能描述:ocr识别管理
 *
 * @author: yfl
 * @date: 2025/8/1 10:10
 */
@Slf4j
@Service
public class OcrServiceImpl implements OcrService {


    @Autowired
    private ApiUrlUtil apiUrlUtil;

    /**
     * 功能描述: ocr识别图片中的文字
     *
     * @param: [multipartFile]
     * @author: yfl
     * @date: 2025/8/1 11:08
     */
    @Override
    public Msg ocrRecognizeImage(MultipartFile multipartFile) {
        return OCRUtils.ocrRecognizeImage(multipartFile);
    }

    /**
     * 功能描述: ocr识别pdf中的文字,返回文本
     *
     * @param: [multipartFile]
     * @author: yfl
     * @date: 2025/8/1 11:37
     */
    @Override
    public Msg extractPdfText(MultipartFile multipartFile) {
        File file = null;
        try {
            String ocrApiUrl = apiUrlUtil.getOcrApiUrl();
            if (StringUtils.isBlank(ocrApiUrl)) {
                return Msg.fail("ocrApiUrl参数未配置!");
            }
             file = FileFormatUtil.multipartFileToFile(multipartFile);
            // 执行请求，文件以form-data形式传参
            String result = HttpRequest.post(ocrApiUrl + "/extract_pdf_text")
                    // 移除错误的Content-Type头，工具会自动生成multipart/form-data头
                    .form("file", file) // 传递文件流和文件名
                    .timeout(200000) // 超时，毫秒
                    .execute()
                    .body();

            if (StringUtils.isBlank(result)) {
                return Msg.fail("OCR识别失败!");
            }
            // 返回结果转json
            JSONObject jsonObject = JSONUtil.parseObj(result);
            if (!jsonObject.getBool("success")) {
                return Msg.sucess(jsonObject.getStr("error"));
            }
            return Msg.sucess(jsonObject.getStr("extractedText"));
        } catch (IOException e) {
            Log.error("OCR识别异常", e);
            return Msg.fail("OCR识别异常!");
        }finally {
            // 关键改进：在finally块中确保文件被删除，无论成功失败都会执行
            if (file != null && file.exists()) {
                // 尝试删除文件
                boolean deleted = file.delete();
                if (!deleted) {
                    // 若删除失败，记录警告日志
                    log.warn("临时文件删除失败，路径：{}", file.getAbsolutePath());
                    // 作为备选方案，设置延迟删除（防止当前有进程占用文件）
                    file.deleteOnExit();
                }
            }
        }
    }


    /**
     * 功能描述: ocr识别pdf中的文字,url传参
     *
     * @param: [extractUrlTextDTO]
     * @author: yfl
     * @date: 2025/8/1 12:20
     */
    @Override
    public Msg extractUrlText(ExtractUrlTextDTO extractUrlTextDTO) {
        String ocrApiUrl = apiUrlUtil.getOcrApiUrl();
        if (StringUtils.isBlank(ocrApiUrl)) {
            return Msg.fail("ocrApiUrl参数未配置!");
        }
        HashMap<String, Object> paramMap = new HashMap<>();
        paramMap.put("url", extractUrlTextDTO.getUrl());
        String result = HttpUtil.post(ocrApiUrl + "/extract_url_text", paramMap);
        if (StringUtils.isBlank(result)) {
            return Msg.fail("OCR识别失败!");
        }
        // 返回结果转json
        JSONObject jsonObject = JSONUtil.parseObj(result);
        if (!jsonObject.getBool("success")) {
            return Msg.sucess(jsonObject.getStr("error"));
        }
        return Msg.sucess(jsonObject.getStr("extractedText"));

    }


}
